#pragma once

#include <iostream>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <cstring>
#include "log.hpp"
#include <string>

Log logbook;
#define pathname "/home/zs"
#define proj_id 0x2303
// 共享内存一般是4096的整数倍(即4KB的整数倍)
// 如果是4097，那么操作系统给的大小是4096*2，但是只能使用4097个字节
#define MEMSIZE 4096
#define FIFO_FILE "./myfifo"
#define MODE 0664

enum
{
    FIFO_CREATE_ERR = 1,
    FIFO_DELETE_ERR,
    FIFO_OPEN_ERR
};

// 共享内存无法实现同步，所以需要借助管道
class Init
{
public:
    Init()
    {
        // 创建管道
        int n = mkfifo(FIFO_FILE,MODE);
        if(n == -1)
        {
            perror("mkfifo");
            exit(FIFO_CREATE_ERR);
        }
    }
    ~Init()
    {
        // 删除管道
        int m = unlink(FIFO_FILE);
        if(m == -1)
        {
            perror("unlink");
            exit(FIFO_DELETE_ERR);
        }
    }
};

// 获取键值(key)
key_t GetKey()
{
    // key_t ftok：创建键值(key)用于创建共享内存等
    // 成功则返回一个非负的键值，用于后续的ipc操作，失败则返回-1
    // key_t:int
    key_t key = ftok(pathname,proj_id);
    if(key < 0)
    {
        logbook(Fatal,"ftok error: %s",strerror(errno));
        exit(1);
    }
    logbook(Info,"ftok success,ley is : %d",key);
    return key;
}
// 获取共享内存
int GetshareMemHelper(int flag)
{
    key_t key = GetKey();

    // 创建共享内存
    // 成功则返回共享内存标示符，否则返回-1
    int shfd = shmget(key,MEMSIZE,flag);
    if(shfd < 0)
    {
        logbook(Fatal,"create share memory error: %s",strerror(errno));
        exit(2);
    }
    logbook(Info,"create share memory success, shmid is: %d",shfd);
    return shfd;
}
int CreateShm()
{
    return GetshareMemHelper(IPC_CREAT | IPC_EXCL | 0666);
}
int GetShm()
{
    return GetshareMemHelper(IPC_CREAT);
}